package com.woniuxy.utils;

import io.jsonwebtoken.*;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

import javax.crypto.spec.SecretKeySpec;
import javax.xml.bind.DatatypeConverter;
import java.security.Key;
import java.util.HashMap;
import java.util.Map;

@Data
@Component
@ConfigurationProperties(prefix = "token.jwt")
public class JWTUtils {
    //Base64对该令牌进行编码
//    @Value("${token.jwt.secret}")
    private String secret;
    /**
     * 加密方式
     */
    private static final SignatureAlgorithm signatureAlgorithm = SignatureAlgorithm.HS256;

    /**
     * 对密钥先base
     *
     * @return
     */
    private Key getkey() {
        System.out.println(secret);
        //将Base64编码后的字符还原为字节数组
        byte[] apiKeySecretBytes = DatatypeConverter.parseBase64Binary(secret);

        //对字节数组按照指定的方式加密

        return new SecretKeySpec(apiKeySecretBytes, signatureAlgorithm.getJcaName());
    }



    /**
     * 生成Token
     * <p>
     * JWT分成3部分：1.头部（header)，2.载荷，3.签证（signature)
     * <p>
     * 加密后这3部分密文的字符位数为：
     * 1.头部（header)：36位，Base64编码
     * 2.载荷（payload)：没准，BASE64编码
     * 3.签证（signature)：43位，将header和payload拼接生成一个字符串，
     * 使用HS256算法和我们提供的密钥（secret,服务器自己提供的一个字符串），
     * 对str进行加密生成最终的JWT
     *
     * @return
     * @throws Exception
     */
    public  String createToken(String username) {

        // 组合header
        Map<String, Object> map = new HashMap<String, Object>();
        map.put("alg", "HS256");
        map.put("typ", "JWT");
        JwtBuilder builder = Jwts.builder();
        builder.setHeaderParams(map);

        // 有效荷载，存放了用户名
        builder.claim("username", username);

        // 生成第三部分（数字签名）
        System.out.println(getkey());
        builder.signWith(SignatureAlgorithm.HS256, getkey());


        String token = builder.compact();
        return token;
    }

    /**
     * 解密Token查看其是否合法
     *包括其中的过期时间和签名是否正确
     * @param token
     * @return
     */
    public  boolean verifyToken(String token) {
        Claims body = null;
        try {
            System.out.println(getkey());
            body = Jwts.parser().setSigningKey(getkey()).parseClaimsJws(token).getBody();
        } catch (ExpiredJwtException e) {
            e.printStackTrace();
            return false;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
        return body != null;
    }

    public  String getUsername(String token) {
        Claims body = null;
        try {

            body = Jwts.parser().setSigningKey(getkey()).parseClaimsJws(token).getBody();
        } catch (ExpiredJwtException e) {
            throw new RuntimeException("超时", e);
        } catch (Exception e) {
            throw new RuntimeException("未知错误", e);
        }
        return (String) body.get("username");
    }

    //解析Jwt
    public Claims getClaimsByToken(String jwt){
        try{
            return Jwts.parser()
                    .setSigningKey(secret) //密钥
                    .parseClaimsJws(jwt)
                    .getBody();
        }catch (Exception e){
            return null;
        }

    }
}
